"""
入口主程序
"""
import torch
from classification.cifar10 import cifar10_run


if __name__ == '__main__':
    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

    # cifar10的图片分类
    cifar10_run(device, path='./data/', is_download=False)
